import copy
from .base_cfg import SEGMENTOR_CFG
from .._base_ import DATASET_CFG_MyGlass_1024x512, DATALOADER_CFG_BS
import os

# deepcopy
SEGMENTOR_CFG = copy.deepcopy(SEGMENTOR_CFG)
# modify dataset config
SEGMENTOR_CFG['dataset'] = DATASET_CFG_MyGlass_1024x512.copy()
SEGMENTOR_CFG['dataset']['rootdir'] = "/home/luoluoluo/data/dataset/elevator/glass_voc"
# modify dataloader config
SEGMENTOR_CFG['dataloader'] = DATALOADER_CFG_BS.copy()
SEGMENTOR_CFG['dataloader']['train']['batch_size'] = 16
SEGMENTOR_CFG['dataloader']['test']['batch_size'] = 1
# modify scheduler config
SEGMENTOR_CFG['scheduler']['max_epochs'] = 300
SEGMENTOR_CFG['save_interval_epochs'] = 10
SEGMENTOR_CFG['eval_interval_epochs'] = 10
SEGMENTOR_CFG['log_interval_iterations'] = 100
# modify other segmentor configs
SEGMENTOR_CFG['num_classes'] = 3
SEGMENTOR_CFG['work_dir'] = 'output/SCTNet_Glass_1024_512'
SEGMENTOR_CFG['logfilepath'] = 'output/SCTNet_Glass_1024_512/SCTNet_Glass_1024_512.log'
SEGMENTOR_CFG['resultsavepath'] = 'output/SCTNet_Glass_1024_512/SCTNet_Glass_1024_512.pkl'
